sort a 2d array based on one column python